package com.wallace.downloadserver.redis

import org.springframework.data.redis.core.RedisTemplate
import java.util.concurrent.TimeUnit

/*-------------------------------------------------------------*/
/**
 * Redis工具类
 */
/*-------------------------------------------------------------*/

/**
 * 指定缓存失效时间
 * @param key 键
 * @param time 时间(秒)
 * @return
 */
fun RedisTemplate<String, Any>.setExpire(key: String, time: Long): Boolean {
    return try {
        if (time > 0) {
            this.expire(key, time, TimeUnit.SECONDS)
        }
        true
    } catch (e: Exception) {
        e.printStackTrace()
        false
    }
}

/**
 * 根据key 获取过期时间
 * @param key 键 不能为null
 * @return 时间(秒) 返回0代表为永久有效
 */
fun RedisTemplate<String, Any>.getExpire(key: String): Long {
    return this.getExpire(key, TimeUnit.SECONDS)
}

/**
 * 判断key是否存在
 * @param key 键
 * @return true 存在 false不存在
 */
fun RedisTemplate<String, Any>.hasKey(key: String): Boolean {
    return try {
        this.hasKey(key)
    } catch (e: Exception) {
        e.printStackTrace()
        false
    }
}

/**
 * 删除缓存
 * @param key 可以传一个值 或多个
 */
fun RedisTemplate<String, Any>.del(vararg key: String) {
    if (key.isNotEmpty()) {
        if (key.size == 1) {
            this.delete(key[0])
        } else {
            this.delete(arrayListOf(*key))
        }
    }
}

/**
 * 普通缓存获取
 * @param key 键
 * @return 值
 */
fun RedisTemplate<String, Any>.getV(key: String): Any? {
    return this.opsForValue().get(key)
}

/**
 * 普通缓存放入
 * @param key 键
 * @param value 值
 * @return true成功 false失败
 */
fun RedisTemplate<String, Any>.setV(key: String, value: Any): Boolean {
    return try {
        this.opsForValue()[key] = value
        true
    } catch (e: Exception) {
        e.printStackTrace()
        false
    }
}

/**
 * 普通缓存放入并设置时间
 * @param key 键
 * @param value 值
 * @param time 时间(秒) time要大于0 如果time小于等于0 将设置无限期
 * @return true成功 false 失败
 */
fun RedisTemplate<String, Any>.setV(key: String, value: Any, time: Long): Boolean {
    return try {
        if (time > 0) {
            this.opsForValue()[key, value, time] = TimeUnit.SECONDS
        } else {
            setV(key, value)
        }
        true
    } catch (e: Exception) {
        e.printStackTrace()
        false
    }
}

/**
 * 递增
 * @param key 键
 * @param delta 要增加几(大于0)
 * @return
 */
fun RedisTemplate<String, Any>.incr(key: String, delta: Long): Long? {
    if (delta < 0) {
        throw RuntimeException("递增因子必须大于0")
    }
    return this.opsForValue().increment(key, delta)
}

/**
 * 递减
 * @param key 键
 * @param delta 要减少几(小于0)
 * @return
 */
fun RedisTemplate<String, Any>.decr(key: String, delta: Long): Long? {
    if (delta < 0) {
        throw RuntimeException("递减因子必须大于0")
    }
    return this.opsForValue().increment(key, -delta)
}

/**
 * HashGet
 * @param key 键 不能为null
 * @param item 项 不能为null
 * @return 值
 */
fun RedisTemplate<String, Any>.hget(key: String, item: String): Any? {
    return this.opsForHash<Any, Any>()[key, item]
}

/**
 * 获取hashKey对应的所有键值
 * @param key 键
 * @return 对应的多个键值
 */
fun RedisTemplate<String, Any>.hmget(key: String): Map<Any, Any> {
    return this.opsForHash<Any, Any>().entries(key)
}

/**
 * HashSet
 * @param key 键
 * @param map 对应多个键值
 * @return true 成功 false 失败
 */
fun RedisTemplate<String, Any>.hmset(key: String, map: Map<String, Any>): Boolean {
    return try {
        this.opsForHash<Any, Any>().putAll(key, map)
        true
    } catch (e: Exception) {
        e.printStackTrace()
        false
    }
}

/**
 * HashSet 并设置时间
 * @param key 键
 * @param map 对应多个键值
 * @param time 时间(秒)
 * @return true成功 false失败
 */
fun RedisTemplate<String, Any>.hmset(key: String, map: Map<String, Any>, time: Long): Boolean {
    return try {
        this.opsForHash<Any, Any>().putAll(key, map)
        if (time > 0) {
            setExpire(key, time)
        }
        true
    } catch (e: Exception) {
        e.printStackTrace()
        false
    }
}

/**
 * 向一张hash表中放入数据,如果不存在将创建
 * @param key 键
 * @param item 项
 * @param value 值
 * @return true 成功 false失败
 */
fun RedisTemplate<String, Any>.hset(key: String, item: String, value: Any): Boolean {
    return try {
        this.opsForHash<Any, Any>().put(key, item, value)
        true
    } catch (e: Exception) {
        e.printStackTrace()
        false
    }
}

/**
 * 向一张hash表中放入数据,如果不存在将创建
 * @param key 键
 * @param item 项
 * @param value 值
 * @param time 时间(秒) 注意:如果已存在的hash表有时间,这里将会替换原有的时间
 * @return true 成功 false失败
 */
fun RedisTemplate<String, Any>.hset(key: String, item: String, value: Any, time: Long): Boolean {
    return try {
        this.opsForHash<Any, Any>().put(key, item, value)
        if (time > 0) {
            setExpire(key, time)
        }
        true
    } catch (e: Exception) {
        e.printStackTrace()
        false
    }
}

/**
 * 删除hash表中的值
 * @param key 键 不能为null
 * @param item 项 可以使多个 不能为null
 */
fun RedisTemplate<String, Any>.hdel(key: String, vararg item: Any?) {
    this.opsForHash<Any, Any>().delete(key, *item)
}

/**
 * 判断hash表中是否有该项的值
 * @param key 键 不能为null
 * @param item 项 不能为null
 * @return true 存在 false不存在
 */
fun RedisTemplate<String, Any>.hHasKey(key: String, item: String): Boolean {
    return this.opsForHash<Any, Any>().hasKey(key, item)
}

/**
 * hash递增 如果不存在,就会创建一个 并把新增后的值返回
 * @param key 键
 * @param item 项
 * @param by 要增加几(大于0)
 * @return
 */
fun RedisTemplate<String, Any>.hincr(key: String, item: String, by: Double): Double {
    return this.opsForHash<Any, Any>().increment(key, item, by)
}

/**
 * hash递减
 * @param key 键
 * @param item 项
 * @param by 要减少记(小于0)
 * @return
 */
fun RedisTemplate<String, Any>.hdecr(key: String, item: String, by: Double): Double {
    return this.opsForHash<Any, Any>().increment(key, item, -by)
}

/**
 * 根据key获取Set中的所有值
 * @param key 键
 * @return
 */
fun RedisTemplate<String, Any>.sGet(key: String): Set<Any>? {
    return try {
        this.opsForSet().members(key)
    } catch (e: Exception) {
        e.printStackTrace()
        null
    }
}

/**
 * 根据value从一个set中查询,是否存在
 * @param key 键
 * @param value 值
 * @return true 存在 false不存在
 */
fun RedisTemplate<String, Any>.sHasKey(key: String, value: Any): Boolean {
    return try {
        this.opsForSet().isMember(key, value) ?: false
    } catch (e: Exception) {
        e.printStackTrace()
        false
    }
}

/**
 * 将数据放入set缓存
 * @param key 键
 * @param values 值 可以是多个
 * @return 成功个数
 */
fun RedisTemplate<String, Any>.sSet(key: String, vararg values: Any?): Long {
    return try {
        this.opsForSet().add(key, *values) ?: -1
    } catch (e: Exception) {
        e.printStackTrace()
        0
    }
}

/**
 * 将set数据放入缓存
 * @param key 键
 * @param time 时间(秒)
 * @param values 值 可以是多个
 * @return 成功个数
 */
fun RedisTemplate<String, Any>.sSetAndTime(key: String, time: Long, vararg values: Any?): Long {
    return try {
        val count = this.opsForSet().add(key, *values) ?: -1L
        if (time > 0) {
            setExpire(key, time)
        }
        count
    } catch (e: Exception) {
        e.printStackTrace()
        0
    }
}

/**
 * 获取set缓存的长度
 * @param key 键
 * @return
 */
fun RedisTemplate<String, Any>.sGetSetSize(key: String): Long {
    return try {
        this.opsForSet().size(key) ?: -1L
    } catch (e: Exception) {
        e.printStackTrace()
        -1L
    }
}

/**
 * 移除值为value的
 * @param key 键
 * @param values 值 可以是多个
 * @return 移除的个数
 */
//    fun RedisTemplate<String, Any>.setRemove(key: String, vararg values: Any?): Long {
//        return try {
//            this.opsForSet().remove(key, *values)
//        } catch (e: Exception) {
//            e.printStackTrace()
//            0
//        }
//    }

/**
 * 获取list缓存的内容
 * @param key 键
 * @param start 开始
 * @param end 结束 0 到 -1代表所有值
 * @return
 */
fun RedisTemplate<String, Any>.lGet(key: String, start: Long, end: Long): List<Any>? {
    return try {
        this.opsForList().range(key, start, end)
    } catch (e: Exception) {
        e.printStackTrace()
        null
    }
}

/**
 * 获取list缓存的长度
 * @param key 键
 * @return
 */
//    fun RedisTemplate<String, Any>.lGetListSize(key: String): Long {
//        return try {
//            this.opsForList().size(key)
//        } catch (e: Exception) {
//            e.printStackTrace()
//            0
//        }
//    }

/**
 * 通过索引 获取list中的值
 * @param key 键
 * @param index 索引 index>=0时， 0 表头，1 第二个元素，依次类推；index<0时，-1，表尾，-2倒数第二个元素，依次类推
 * @return
 */
fun RedisTemplate<String, Any>.lGetIndex(key: String, index: Long): Any? {
    return try {
        this.opsForList().index(key, index)
    } catch (e: Exception) {
        e.printStackTrace()
        null
    }
}

/**
 * 将list放入缓存
 * @param key 键
 * @param value 值
 * @return
 */
fun RedisTemplate<String, Any>.lSet(key: String, value: Any): Boolean {
    return try {
        this.opsForList().rightPush(key, value)
        true
    } catch (e: Exception) {
        e.printStackTrace()
        false
    }
}

/**
 * 将list放入缓存
 * @param key 键
 * @param value 值
 * @param time 时间(秒)
 * @return
 */
fun RedisTemplate<String, Any>.lSet(key: String, value: Any, time: Long): Boolean {
    return try {
        this.opsForList().rightPush(key, value)
        if (time > 0) {
            setExpire(key, time)
        }
        true
    } catch (e: Exception) {
        e.printStackTrace()
        false
    }
}

/**
 * 将list放入缓存
 * @param key 键
 * @param value 值
 * @return
 */
fun RedisTemplate<String, Any>.lSet(key: String, value: List<Any>): Boolean {
    return try {
        this.opsForList().rightPushAll(key, *value.toTypedArray())
        true
    } catch (e: Exception) {
        e.printStackTrace()
        false
    }
}

/**
 * 将list放入缓存
 *
 * @param key 键
 * @param value 值
 * @param time 时间(秒)
 * @return
 */
fun RedisTemplate<String, Any>.lSet(key: String, value: List<Any?>, time: Long): Boolean {
    return try {
        this.opsForList().rightPushAll(key, *value.toTypedArray())
        if (time > 0) {
            setExpire(key, time)
        }
        true
    } catch (e: Exception) {
        e.printStackTrace()
        false
    }
}

/**
 * 根据索引修改list中的某条数据
 * @param key 键
 * @param index 索引
 * @param value 值
 * @return
 */
fun RedisTemplate<String, Any>.lUpdateIndex(key: String, index: Long, value: Any): Boolean {
    return try {
        this.opsForList()[key, index] = value
        true
    } catch (e: Exception) {
        e.printStackTrace()
        false
    }
}

/**
 * 移除N个值为value
 * @param key 键
 * @param count 移除多少个
 * @param value 值
 * @return 移除的个数
 */
//    fun RedisTemplate<String, Any>.lRemove(key: String, count: Long, value: Any?): Long {
//        return try {
//            this.opsForList().remove(key, count, value)
//        } catch (e: Exception) {
//            e.printStackTrace()
//            0
//        }
//    }